from .mutr import build
from .tlggn.tlggn import build_tlggn
from .tlggn.tlggnv1 import build_tlggnv1

def build_model(args):
    if args.me == 'tlggn':
        return build_tlggn(args)
    elif args.me == 'tlggnv1':
        return build_tlggnv1(args)
    return build(args)
